1066B - Heaters - CodeForces Solution

#include <iostream>
#include <math.h>
#include <vector>
using namespace std;

int main(){
    int n, r;
    int a;
    cin >> n >> r;
    int arr[n];
    int ans = 0;
    int aux;
    vector<int>calentador;
    for(int i=0; i<n; i++){
        cin >> a;
        arr[i] = -1;
        if(a == 1){
            calentador.push_back(i);
        }
    }
    for(int i=0; i<calentador.size(); i++){
        aux = r;
        while(aux > 0){
            if(calentador[i]-aux+1 >= 0){
                arr[calentador[i]-aux+1] = calentador[i];
            }
            if(calentador[i]+aux-1 <= n-1){
                arr[calentador[i]+aux-1] = calentador[i];
            }
            aux--;
        }
        arr[calentador[i]] = calentador[i];
    }
    int ui = 0;
    while(ui<n){
        if(arr[ui] == -1){
            ans = -1;
            break;
        }
        ans++;
        ui = arr[ui] + r;

    }
    /*for(int j=0; j<n; j++){
        if(arr[j] == -1){
            ans = arr[j];
            break;
        }
        ans++;

        for(int i=0; i<calentador.size(); i++){
            if(j < calentador[i]){
                break;
            }
            if(j == calentador[i]){
                ans++;
                j += (r-1);
            }
        }
    }*/
    cout << ans;
}